import numpy as np
from util.TDataGenerator import DataGenerator as DataGenerator

if __name__ == "__main__":
    num_rows = 5  # 指定行数
    num_cols = 3  # 指定列数

    generator = DataGenerator(num_rows, num_cols)
    filename = 'train_output/trainNum01.txt'  # 文件名

    # 写入数据到 main.py
    generator.write_data_to_file(filename)

    # 从 main.py 读取数据并打印
    print("从 main.py 中读取的数据:")
    generator.read_data_from_file(filename)